import torch
# 終端 watch -n 1 nvidia-smi
print('蟒炬版本',torch.__version__)
print(f'凡 {torch.cuda.device_count()} 張顯牒')
if torch.cuda.is_available():
	臨時 = '可用，版本 '+ torch.version.cuda +'。'
else:
	臨時 = '不可用。'
print('CUDA',臨時)
牒數 = torch.cuda.device_count()
print('顯牒凡', 牒數,'張：')
for 牒 in range(牒數):
	print(f'[{牒}]《{torch.cuda.get_device_name(牒)}》算力 {torch.cuda.get_device_capability(牒)}，屬性 {torch.cuda.get_device_properties(牒)}')
# torch.cuda.set_device(0)
print('當歬顯牒爲', torch.cuda.current_device())